package com.huabo.utils;

import org.apache.log4j.Logger;

import java.io.UnsupportedEncodingException;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.util.Arrays;

public class MD5Utils {

	private static final String HEX_NUMS_STR = "0123456789ABCDEF";
	private static final Integer SALT_LENGTH = 12;
	private static final Logger LOG = Logger.getLogger(MD5Utils.class);

	/**
	 * 将16进制字符串转换成字节数组
	 * 
	 * @param hex
	 * @return
	 */
	public static byte[] hexStringToByte(String hex) {
		int len = hex.length() / 2;
		byte[] result = new byte[len];
		char[] hexChars = hex.toCharArray();
		for (int i = 0; i < len; i++) {
			int pos = i * 2;
			result[i] = (byte) (HEX_NUMS_STR.indexOf(hexChars[pos]) << 4 | HEX_NUMS_STR
					.indexOf(hexChars[pos + 1]));
		}
		return result;
	}

	/**
	 * 将指定byte数组转换成16进制字符串
	 * 
	 * @param b
	 * @return
	 */
	public static String byteToHexString(byte[] b) {
		StringBuilder hexString = new StringBuilder();
		for (int i = 0; i < b.length; i++) {
			String hex = Integer.toHexString(b[i] & 0xFF);
			if (hex.length() == 1) {
				hex = '0' + hex;
			}
			hexString.append(hex.toUpperCase());
		}
		return hexString.toString();
	}

	/**
	 * 验证口令是否合法
	 * 
	 * @param password
	 * @param passwordInDb
	 * @return
	 * @throws NoSuchAlgorithmException
	 * @throws UnsupportedEncodingException
	 */
	public static boolean validPassword(String password, String passwordInDb) {

		try {
			// 将16进制字符串格式口令转换成字节数组
			byte[] pwdInDb = hexStringToByte(passwordInDb);
			// 声明盐变量
			byte[] salt = new byte[SALT_LENGTH];
			// 将盐从数据库中保存的口令字节数组中提取出来
			System.arraycopy(pwdInDb, 0, salt, 0, SALT_LENGTH);
			// 创建消息摘要对象
			MessageDigest md = MessageDigest.getInstance("MD5");
			// 将盐数据传入消息摘要对象
			md.update(salt);
			// 将口令的数据传给消息摘要对象
			md.update(password.getBytes("UTF-8"));
			// 生成输入口令的消息摘要
			byte[] digest = md.digest();
			// 声明一个保存数据库中口令消息摘要的变量
			byte[] digestInDb = new byte[pwdInDb.length - SALT_LENGTH];
			// 取得数据库中口令的消息摘要
			System.arraycopy(pwdInDb, SALT_LENGTH, digestInDb, 0,
					digestInDb.length);
			// 比较根据输入口令生成的消息摘要和数据库中消息摘要是否相同
			if (Arrays.equals(digest, digestInDb)) {
				// 口令正确返回口令匹配消息
				return true;
			} 
			return false;
			
		} catch (NoSuchAlgorithmException e) {
			LOG.error(e.getMessage(), e);
		} catch (UnsupportedEncodingException e) {
			LOG.error(e.getMessage(), e);
		}
		return false;
	}
	
	public static String encryptWithoutSalt(String plainText) {
		try {
			// 声明消息摘要对象
			MessageDigest md = null;
			// 创建消息摘要
			md = MessageDigest.getInstance("MD5");
			// 将口令的数据传给消息摘要对象
			md.update(plainText.getBytes("UTF-8"));
			// 将字节数组格式加密后的口令转化为16进制字符串格式的口令
			return byteToHexString(md.digest());
		} catch (NoSuchAlgorithmException e) {
			LOG.error(e.getMessage(), e);
		} catch (UnsupportedEncodingException e) {
			LOG.error(e.getMessage(), e);
		}
		return null;
	}
	
	public static boolean validateWithoutSalt(String plainText, String encrypedText) {
		String newEncrypt = encryptWithoutSalt(plainText);
		return newEncrypt.equals(encrypedText);
	}

	/**
	 * 对字符串进行加密
	 * 
	 * @param password
	 * @return
	 * @throws NoSuchAlgorithmException
	 * @throws UnsupportedEncodingException
	 */
	public static String encrypt(String password) {
		try {
			// 声明加密后的口令数组变量
			byte[] pwd = null;
			// 随机数生成器
			SecureRandom random = new SecureRandom();
			// 声明盐数组变量
			byte[] salt = new byte[SALT_LENGTH];
			// 将随机数放入盐变量中
			random.nextBytes(salt);

			// 声明消息摘要对象
			MessageDigest md = null;
			// 创建消息摘要
			md = MessageDigest.getInstance("MD5");
			// 将盐数据传入消息摘要对象
			md.update(salt);
			// 将口令的数据传给消息摘要对象
			md.update(password.getBytes("UTF-8"));
			// 获得消息摘要的字节数组
			byte[] digest = md.digest();

			// 因为要在口令的字节数组中存放盐，所以加上盐的字节长度
			pwd = new byte[digest.length + SALT_LENGTH];
			// 将盐的字节拷贝到生成的加密口令字节数组的前12个字节，以便在验证口令时取出盐
			System.arraycopy(salt, 0, pwd, 0, SALT_LENGTH);
			// 将消息摘要拷贝到加密口令字节数组从第13个字节开始的字节
			System.arraycopy(digest, 0, pwd, SALT_LENGTH, digest.length);
			// 将字节数组格式加密后的口令转化为16进制字符串格式的口令
			return byteToHexString(pwd);
		} catch (NoSuchAlgorithmException e) {
			LOG.error(e.getMessage(), e);
		} catch (UnsupportedEncodingException e) {
			LOG.error(e.getMessage(), e);
		}
		return null;

	}
	
	public static String getSalt(String passwordInDb) {
		byte[] pwdInDb = hexStringToByte(passwordInDb);
		// 声明盐变量
		byte[] salt = new byte[SALT_LENGTH];
		// 将盐从数据库中保存的口令字节数组中提取出来
		System.arraycopy(pwdInDb, 0, salt, 0, SALT_LENGTH);
		
		return byteToHexString(salt);
	}
	
	public static String encryptTo32digit(String sourceStr) {
        String result = "";
        try {
            MessageDigest md = MessageDigest.getInstance("MD5");
            md.update(sourceStr.getBytes());
            byte b[] = md.digest();
            int i;
            StringBuffer buf = new StringBuffer("");
            for (int offset = 0; offset < b.length; offset++) {
                i = b[offset];
                if (i < 0) {
                    i += 256;
                }
                if (i < 16) {
                    buf.append("0");
                }
                buf.append(Integer.toHexString(i));
            }
            result = buf.toString();
        } catch (NoSuchAlgorithmException e) {
        	LOG.error(e.getMessage(), e);
        }
        return result;
    }

}
